package com.ghp.admin.interceptor;


import cn.hutool.core.util.StrUtil;
import com.ghp.common.model.vo.UserLoginVO;
import com.ghp.common.utils.BeanConvertorUtils;
import com.ghp.common.utils.JwtGeneratorUtils;
import com.ghp.common.utils.ThreadLocalUtils;
import io.jsonwebtoken.Claims;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import static com.ghp.common.constants.RedisConstants.LOGIN_USER_KEY;
import static com.ghp.common.constants.RedisConstants.LOGIN_USER_TTL;
import static com.ghp.common.constants.SystemConstants.TOKEN_HEADER_KEY;

/**
 * @author ghp
 * @title 刷新 Redis 中 token 的拦截器，同时判断token是否合法
 * @description
 */
@Slf4j
public class RefreshTokenInterceptor implements HandlerInterceptor {

    private StringRedisTemplate stringRedisTemplate;

    public RefreshTokenInterceptor(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }

    /**
     * 拦截所有请求
     * @param request
     * @param response
     * @param handler
     * @return
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        // 获取token，并判断token是否存在
        String token = request.getHeader(TOKEN_HEADER_KEY);
        if (StrUtil.isBlank(token)) {
            // token不存在，说明当前用户未登录，不需要刷新直接放行（后面的登录拦截器会拦截）
            return true;
        }
        // token存在，说明当前用户是存在的，判断token是否合法
        Claims claims = null;
        try {
            claims = JwtGeneratorUtils.parseJWT(token);
        } catch (Exception e) {
            // token超时或非法，直接放心，后面登录拦截器会进行校验
            return true;
        }
        // 判断用户是否存在
        String userId = claims.getSubject();
        String key = LOGIN_USER_KEY + userId;
        Map<Object, Object> userMap = stringRedisTemplate.opsForHash().entries(key);
        if (userMap.isEmpty()) {
            log.error("key={}, value={}。可能是Redis的key过期了", key, userMap);
            return true;
        }
        // 用户存在，则将用户信息保存到ThreadLocal中，方便后续逻辑处理，比如：方便获取和使用用户信息，Redis获取用户信息是具有侵入性的
        UserLoginVO userLoginVO = null;
        try {
            userLoginVO = BeanConvertorUtils.mapToBean(userMap, UserLoginVO.class);
        } catch (Exception e) {
            log.error("Map转Bean失败 \n {}", e.getStackTrace());
            return true;
        }
        ThreadLocalUtils.setCurrentId(userLoginVO.getId());
        // 刷新token有效期
        stringRedisTemplate.expire(token, LOGIN_USER_TTL, TimeUnit.SECONDS);
        return true;
    }
}
